package org.hao.samples.interceptor;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import org.hao.samples.model.Student;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

public class LoginInterceptor implements HandlerInterceptor {

	@Override
	public void afterCompletion(HttpServletRequest arg0, HttpServletResponse arg1, Object arg2, Exception arg3)
			throws Exception {
		System.out.println("----------------LoginInterceptor.afterCompletion()--------------------");

	}

	@Override
	public void postHandle(HttpServletRequest arg0, HttpServletResponse arg1, Object arg2, ModelAndView arg3)
			throws Exception {
		System.out.println("----------------LoginInterceptor.postHandle()--------------------");

	}

	@Override
	public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object arg2) throws Exception {
		HttpSession session = request.getSession();
		Student examUser = (Student) session.getAttribute("examUser");
		Integer role = (Integer) session.getAttribute("role");
		if (examUser != null) {
			request.setAttribute("role", role);
			return true;
		} else {
			response.sendRedirect(request.getContextPath() + "/login");
		}
		return false;
	}

}
